package com.wl.wuliu.weixin;

import cn.hutool.json.JSONException;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.wl.wuliu.entity.SysUser;
import com.wl.wuliu.exception.GlobalExceptionHandler;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.net.ssl.HttpsURLConnection;
import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLSocketFactory;
import javax.net.ssl.TrustManager;
import java.io.*;
import java.net.ConnectException;
import java.net.URL;

/**
 * @Author: gutaozhen
 * @Date: 2024/11/18
 */
public class WeixinHttpUtil {
    public static final Logger log = LoggerFactory.getLogger(GlobalExceptionHandler.class);

    //公众号appid
    public final static String APP_ID = "wx0f51e44efc797407";

    //公众号密钥
    public final static String SECRET = "a6feed12c94e0d0678babc423ae576a4";

    // 凭证获取（token）
    public final static String TOKEN_URL = "https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid=APPID&secret=SECRET";

    // 用户信息获取
    public final static String USER_URL = "https://api.weixin.qq.com/sns/userinfo?access_token=ACCESS_TOKEN&openid=OPENID&lang=zh_CN";

    // 获取网页授权token
    public final static String ACCESS_TOKEN_URL = "https://api.weixin.qq.com/sns/oauth2/access_token?appid=APPID&secret=SECRET&code=CODE&grant_type=authorization_code";

    /**
     * 获取接口访问凭证
     * @return
     */
    public static Token getToken() {
        Token token = null;

        // 拼接请求地址
        String requestUrl = TOKEN_URL;
        requestUrl = requestUrl.replace("APPID", APP_ID).replace("SECRET", SECRET);
        // 发起GET请求获取凭证
        JSONObject jsonObject = httpsRequest(requestUrl, "GET", null);

        if (null != jsonObject) {
            try {
                token = new Token();
                token.setAccessToken(jsonObject.getStr("access_token"));
                token.setExpiresIn(jsonObject.getInt("expires_in"));
            } catch (JSONException e) {
                token = null;
                // 获取token失败
                log.error("获取token失败 errcode:{} errmsg:{}", jsonObject.getInt("errcode"), jsonObject.getStr("errmsg"));
            }
        }
        return token;
    }

    /**
     * 根据code获取授权的token和openid
     * @return
     */
    public static AccessToken getAccessToken(String code) {
        AccessToken accessToken = null;

        // 拼接请求地址
        String requestUrl = ACCESS_TOKEN_URL;
        requestUrl = requestUrl.replace("APPID", APP_ID).replace("SECRET", SECRET).replace("CODE", code);
        // 发起GET请求获取凭证
        JSONObject jsonObject = httpsRequest(requestUrl, "GET", null);

        if (null != jsonObject) {
            try {
                accessToken = new AccessToken();
                accessToken.setAccessToken(jsonObject.getStr("access_token"));
                accessToken.setExpiresIn(jsonObject.getInt("expires_in"));
                accessToken.setRefreshToken(jsonObject.getStr("refresh_token"));
                accessToken.setOpenId(jsonObject.getStr("openid"));
                accessToken.setScope(jsonObject.getStr("scope"));
                accessToken.setUnionid(jsonObject.getStr("unionid"));
            } catch (Exception e) {
                accessToken = null;
                // 获取token失败
                log.error("获取网页授权token失败 errcode:{} errmsg:{}", jsonObject.getInt("errcode"), jsonObject.getStr("errmsg"));
            }
        }
        return accessToken;
    }

    /**
     * 获取用户信息
     *
     * @param openId 用户标识
     * @return WeixinUserInfo
     */
    public static SysUser getUserInfo(String accessToken, String openId) {
        SysUser weixinUserInfo = null;
        // 拼接请求地址
        String requestUrl = USER_URL;
        requestUrl = requestUrl.replace("ACCESS_TOKEN", accessToken).replace("OPENID", openId);
        // 获取用户信息
        JSONObject jsonObject = WeixinHttpUtil.httpsRequest(requestUrl, "GET", null);

        if (null != jsonObject) {
            try {
                weixinUserInfo = new SysUser();
                // 用户的标识
                weixinUserInfo.setOpenId(jsonObject.getStr("openid"));
                // 关注状态（1是关注，0是未关注），未关注时获取不到其余信息
//                weixinUserInfo.setSubscribe(jsonObject.getInt("subscribe"));
                // 用户关注时间
//                weixinUserInfo.setSubscribeTime(jsonObject.getStr("subscribe_time"));
                // 昵称
                weixinUserInfo.setNickName(jsonObject.getStr("nickname"));
                // 用户的性别（1是男性，2是女性，0是未知）
                weixinUserInfo.setSex(jsonObject.getStr("sex"));
                // 用户所在国家
                weixinUserInfo.setCountry(jsonObject.getStr("country"));
                // 用户所在省份
                weixinUserInfo.setProvince(jsonObject.getStr("province"));
                // 用户所在城市
                weixinUserInfo.setCity(jsonObject.getStr("city"));

                // 用户头像
                weixinUserInfo.setHeadimgurl(jsonObject.getStr("headimgurl"));
            } catch (Exception e) {
                if (0 == jsonObject.getInt("subscribe")) {
                    log.error("用户{}已取消关注", openId);
                } else {
                    int errorCode = jsonObject.getInt("errcode");
                    String errorMsg = jsonObject.getStr("errmsg");
                    log.error("获取用户信息失败 errcode:{} errmsg:{}", errorCode, errorMsg);
                }
            }
        }
        return weixinUserInfo;
    }

    /**
     * 发送https请求
     * @param requestUrl 请求地址
     * @param requestMethod 请求方式（GET、POST）
     * @param outputStr 提交的数据
     * @return JSONObject(通过JSONObject.get(key)的方式获取json对象的属性值)
     */
    public static JSONObject httpsRequest(String requestUrl, String requestMethod, String outputStr) {
        JSONObject jsonObject = new JSONObject();
        try {
            // 创建SSLContext对象，并使用我们指定的信任管理器初始化
            TrustManager[] tm = { new MyX509TrustManager() };
            SSLContext sslContext = SSLContext.getInstance("SSL", "SunJSSE");
            sslContext.init(null, tm, new java.security.SecureRandom());
            // 从上述SSLContext对象中得到SSLSocketFactory对象
            SSLSocketFactory ssf = sslContext.getSocketFactory();

            URL url = new URL(requestUrl);
            HttpsURLConnection conn = (HttpsURLConnection) url.openConnection();
            conn.setSSLSocketFactory(ssf);

            conn.setDoOutput(true);
            conn.setDoInput(true);
            conn.setUseCaches(false);
            // 设置请求方式（GET/POST）
            conn.setRequestMethod(requestMethod);

            // 当outputStr不为null时向输出流写数据
            if (null != outputStr) {
                OutputStream outputStream = conn.getOutputStream();
                // 注意编码格式
                outputStream.write(outputStr.getBytes("UTF-8"));
                outputStream.close();
            }

            // 从输入流读取返回内容
            InputStream inputStream = conn.getInputStream();
            InputStreamReader inputStreamReader = new InputStreamReader(inputStream, "utf-8");
            BufferedReader bufferedReader = new BufferedReader(inputStreamReader);
            String str;
            StringBuffer buffer = new StringBuffer();
            while ((str = bufferedReader.readLine()) != null) {
                buffer.append(str);
            }

            // 释放资源
            bufferedReader.close();
            inputStreamReader.close();
            inputStream.close();
            conn.disconnect();
            jsonObject = JSONUtil.parseObj(buffer.toString());
        } catch (ConnectException ce) {
            log.error("连接超时：{}", ce);
        } catch (Exception e) {
            log.error("https请求异常：{}", e);
        }
        return jsonObject;
    }

    /**
     * URL编码（utf-8）
     *
     * @param source
     * @return
     */
    public static String urlEncodeUTF8(String source) {
        String result = source;
        try {
            result = java.net.URLEncoder.encode(source, "utf-8");
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        }
        return result;
    }

    /**
     * 根据内容类型判断文件扩展名
     *
     * @param contentType
     *            内容类型
     * @return
     */
    public static String getFileExt(String contentType) {
        String fileExt = "";
        if ("image/jpeg".equals(contentType))
            fileExt = ".jpg";
        else if ("audio/mpeg".equals(contentType))
            fileExt = ".mp3";
        else if ("audio/amr".equals(contentType))
            fileExt = ".amr";
        else if ("video/mp4".equals(contentType))
            fileExt = ".mp4";
        else if ("video/mpeg4".equals(contentType))
            fileExt = ".mp4";
        return fileExt;
    }

}
